package com.zjsru.plan2025.oneday;

/**
 * 3392. 统计符合条件长度为 3 的子数组数目
 *
 * @Author: cookLee
 * @Date: 2025-04-27
 */
public class CountSubarrays {

    /**
     * 主
     * \
     * 输入：nums = [1,2,1,4,1]
     * <p>
     * 输出：1
     * <p>
     * 解释：
     * <p>
     * 只有子数组 [1,4,1] 包含 3 个元素且第一个和第三个数字之和是中间数字的一半。number.
     * \
     * 输入：nums = [1,1,1]
     * <p>
     * 输出：0
     * <p>
     * 解释：
     * <p>
     * [1,1,1] 是唯一长度为 3 的子数组，但第一个数和第三个数的和不是第二个数的一半。
     * \
     *
     * @param args args
     */
    public static void main(String[] args) {
        CountSubarrays countSubarrays = new CountSubarrays();
        int[] nums = new int[]{1, 2, 1, 4, 1};
        System.out.println(countSubarrays.countSubarrays(nums));
    }

    public int countSubarrays(int[] nums) {
        int len = nums.length;
        int ans = 0;
        for (int i = 1; i < len - 1; i++) {
            int pre = nums[i - 1];
            int after = nums[i + 1];
            if (2 * (pre + after) == nums[i]) {
                ans++;
            }
        }
        return ans;
    }

}
